name='origin'

ngpu=2
k=7
batch_size=256
nproc_per_node=2
model='ResNet18'
loss_type='sim'
advtrain_type='Rep'
attack_type='linf'
regularize_to='other'
attack_to='other'
train_type='contrastive'
dataset='cifar-10'

cd ./src
python -m torch.distributed.launch --nproc_per_node=${nproc_per_node} rocl_train.py --ngpu ${ngpu} --batch-size=${batch_size} --model=${model} --k=${k} --loss_type=${loss_type} --advtrain_type=${advtrain_type} --attack_type=${attack_type} --name=${name} --regularize_to=${regularize_to} --attack_to=${attack_to} --train_type=${train_type} --dataset=${dataset}